try:
    import torch
    print('PyTorch已安装，版本:', torch.__version__)
    print('CUDA是否可用:', torch.cuda.is_available())
    if torch.cuda.is_available():
        print('GPU数量:', torch.cuda.device_count())
        print('当前GPU:', torch.cuda.get_device_name(0))
        print('CUDA版本:', torch.version.cuda)
except ImportError:
    print('PyTorch未安装')